/*
===========================================================================
Copyright (C) 1999-2005 Id Software, Inc.
 
This file is part of Quake III Arena source code.
 
Quake III Arena source code is free software; you can redistribute it
and/or modify it under the terms of the GNU General Public License as
published by the Free Software Foundation; either version 2 of the License,
or (at your option) any later version.
 
Quake III Arena source code is distributed in the hope that it will be
useful, but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
 
You should have received a copy of the GNU General Public License
along with Foobar; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
===========================================================================
*/
#include <float.h>

#include "../snd_local.h"
#include "win_local.h"

HRESULT (WINAPI *pDirectSoundCreate)(GUID FAR *lpGUID, LPDIRECTSOUND FAR *lplpDS, IUnknown FAR *pUnkOuter);
#define iDirectSoundCreate(a,b,c)	pDirectSoundCreate(a,b,c)

#define SECONDARY_BUFFER_SIZE	0x10000


static qboolean	dsound_init;
static int		sample16;
static DWORD	gSndBufSize;
static DWORD	locksize;
static LPDIRECTSOUND pDS;
static LPDIRECTSOUNDBUFFER pDSBuf, pDSPBuf;
static HINSTANCE hInstDS;


static const char *DSoundError( int error )
{
    switch ( error )
    {
    case DSERR_BUFFERLOST:
        return "DSERR_BUFFERLOST";
    case DSERR_INVALIDCALL:
        return "DSERR_INVALIDCALLS";
    case DSERR_INVALIDPARAM:
        return "DSERR_INVALIDPARAM";
    case DSERR_PRIOLEVELNEEDED:
        return "DSERR_PRIOLEVELNEEDED";
    }

    return "unknown";
}

/*
==================
SNDDMA_Shutdown
==================
*/
void SNDDMA_Shutdown( void )
{
    Com_DPrintf( "Shutting down sound system\n" );

    if ( pDS )
    {
        Com_DPrintf( "Destroying DS buffers\n" );
        if ( pDS )
        {
            Com_DPrintf( "...setting NORMAL coop level\n" );
            pDS->lpVtbl->SetCooperativeLevel( pDS, g_wv.hWnd, DSSCL_PRIORITY );
        }

        if ( pDSBuf )
        {
            Com_DPrintf( "...stopping and releasing sound buffer\n" );
            pDSBuf->lpVtbl->Stop( pDSBuf );
            pDSBuf->lpVtbl->Release( pDSBuf );
        }

        // only release primary buffer if it's not also the mixing buffer we just released
        if ( pDSPBuf && ( pDSBuf != pDSPBuf ) )
        {
            Com_DPrintf( "...releasing primary buffer\n" );
            pDSPBuf->lpVtbl->Release( pDSPBuf );
        }
        pDSBuf = NULL;
        pDSPBuf = NULL;

        dma.buffer = NULL;

        Com_DPrintf( "...releasing DS object\n" );
        pDS->lpVtbl->Release( pDS );
    }

    if ( hInstDS )
    {
        Com_DPrintf( "...freeing DSOUND.DLL\n" );
        FreeLibrary( hInstDS );
        hInstDS = NULL;
    }

    pDS = NULL;
    pDSBuf = NULL;
    pDSPBuf = NULL;
    dsound_init = qfalse;
    memset ((void *)&dma, 0, sizeof (dma));
    CoUninitialize( );
}

/*
==================
SNDDMA_Init
 
Initialize direct sound
Returns false if failed
==================
*/
qboolean SNDDMA_Init(void)
{

    memset ((void *)&dma, 0, sizeof (dma));
    dsound_init = 0;

    CoInitialize(NULL);

    if ( !SNDDMA_InitDS () )
    {
        return qfalse;
    }

    dsound_init = qtrue;

    Com_DPrintf("Completed successfully\n" );

    return qtrue;
}

#undef DEFINE_GUID

#define DEFINE_GUID(name, l, w1, w2, b1, b2, b3, b4, b5, b6, b7, b8) \
        EXTERN_C const GUID name \
                = { l, w1, w2, { b1, b2,  b3,  b4,  b5,  b6,  b7,  b8 } }

// DirectSound Component GUID {47D4D946-62E8-11CF-93BC-444553540000}
DEFINE_GUID(CLSID_DirectSound, 0x47d4d946, 0x62e8, 0x11cf, 0x93, 0xbc, 0x44, 0x45, 0x53, 0x54, 0x0, 0x0);

// DirectSound 8.0 Component GUID {3901CC3F-84B5-4FA4-BA35-AA8172B8A09B}
DEFINE_GUID(CLSID_DirectSound8, 0x3901cc3f, 0x84b5, 0x4fa4, 0xba, 0x35, 0xaa, 0x81, 0x72, 0xb8, 0xa0, 0x9b);

DEFINE_GUID(IID_IDirectSound8, 0xC50A7E93, 0xF395, 0x4834, 0x9E, 0xF6, 0x7F, 0xA9, 0x9D, 0xE5, 0x09, 0x66);
DEFINE_GUID(IID_IDirectSound, 0x279AFA83, 0x4981, 0x11CE, 0xA5, 0x21, 0x00, 0x20, 0xAF, 0x0B, 0xE5, 0x60);


int SNDDMA_InitDS ()
{
    HRESULT			hresult;
    DSBUFFERDESC	dsbuf;
    DSBCAPS			dsbcaps;
    WAVEFORMATEX	format;
    int				use8;

    Com_Printf( "Initializing DirectSound\n");

    use8 = 1;
    // Create IDirectSound using the primary sound device
    if( FAILED( hresult = CoCreateInstance(&CLSID_DirectSound8, NULL, CLSCTX_INPROC_SERVER, &IID_IDirectSound8, (void **)&pDS)))
    {
        use8 = 0;
        if( FAILED( hresult = CoCreateInstance(&CLSID_DirectSound, NULL, CLSCTX_INPROC_SERVER, &IID_IDirectSound, (void **)&pDS)))
        {
            Com_Printf ("failed\n");
            SNDDMA_Shutdown ();
            return qfalse;
        }
    }

    hresult = pDS->lpVtbl->Initialize( pDS, NULL);

    Com_DPrintf( "ok\n" );

    Com_DPrintf("...setting DSSCL_PRIORITY coop level: " );

    if ( DS_OK != pDS->lpVtbl->SetCooperativeLevel( pDS, g_wv.hWnd, DSSCL_PRIORITY ) )
    {
        Com_Printf ("failed\n");
        SNDDMA_Shutdown ();
        return qfalse;
    }
    Com_DPrintf("ok\n" );


    // create the secondary buffer we'll actually work with
    dma.channels = 2;
    dma.samplebits = 16;

//	if (s_khz->integer == 44)
//		dma.speed = 44100;
//	else if (s_khz->integer == 22)
//		dma.speed = 22050;
//	else
//		dma.speed = 11025;

    dma.speed = 22050;
    memset (&format, 0, sizeof(format));
    format.wFormatTag = WAVE_FORMAT_PCM;
    format.nChannels = dma.channels;
    format.wBitsPerSample = dma.samplebits;
    format.nSamplesPerSec = dma.speed;
    format.nBlockAlign = format.nChannels * format.wBitsPerSample / 8;
    format.cbSize = 0;
    format.nAvgBytesPerSec = format.nSamplesPerSec*format.nBlockAlign;

    memset (&dsbuf, 0, sizeof(dsbuf));
    dsbuf.dwSize = sizeof(DSBUFFERDESC);

    // Micah: take advantage of 2D hardware.if available.
    dsbuf.dwFlags = DSBCAPS_LOCHARDWARE;
    if (use8)
    {
        dsbuf.dwFlags |= DSBCAPS_GETCURRENTPOSITION2;
    }
    dsbuf.dwBufferBytes = SECONDARY_BUFFER_SIZE;
    dsbuf.lpwfxFormat = &format;

    memset(&dsbcaps, 0, sizeof(dsbcaps));
    dsbcaps.dwSize = sizeof(dsbcaps);

    Com_DPrintf( "...creating secondary buffer: " );
    if (DS_OK == pDS->lpVtbl->CreateSoundBuffer(pDS, &dsbuf, &pDSBuf, NULL))
    {
        Com_Printf( "locked hardware.  ok\n" );
    }
    else
    {
        // Couldn't get hardware, fallback to software.
        dsbuf.dwFlags = DSBCAPS_LOCSOFTWARE;
        if (use8)
        {
            dsbuf.dwFlags |= DSBCAPS_GETCURRENTPOSITION2;
        }
        if (DS_OK != pDS->lpVtbl->CreateSoundBuffer(pDS, &dsbuf, &pDSBuf, NULL))
        {
            Com_Printf( "failed\n" );
            SNDDMA_Shutdown ();
            return qfalse;
        }
        Com_DPrintf( "forced to software.  ok\n" );
    }

    // Make sure mixer is active
    if ( DS_OK != pDSBuf->lpVtbl->Play(pDSBuf, 0, 0, DSBPLAY_LOOPING) )
    {
        Com_Printf ("*** Looped sound play failed ***\n");
        SNDDMA_Shutdown ();
        return qfalse;
    }

    // get the returned buffer size
    if ( DS_OK != pDSBuf->lpVtbl->GetCaps (pDSBuf, &dsbcaps) )
    {
        Com_Printf ("*** GetCaps failed ***\n");
        SNDDMA_Shutdown ();
        return qfalse;
    }

    gSndBufSize = dsbcaps.dwBufferBytes;

    dma.channels = format.nChannels;
    dma.samplebits = format.wBitsPerSample;
    dma.speed = format.nSamplesPerSec;
    dma.samples = gSndBufSize/(dma.samplebits/8);
    dma.submission_chunk = 1;
    dma.buffer = NULL;			// must be locked first

    sample16 = (dma.samplebits/8) - 1;

    SNDDMA_BeginPainting ();
    if (dma.buffer)
        memset(dma.buffer, 0, dma.samples * dma.samplebits/8);
    SNDDMA_Submit ();
    return 1;
}
/*
==============
SNDDMA_GetDMAPos
 
return the current sample position (in mono samples read)
inside the recirculating dma buffer, so the mixing code will know
how many sample are required to fill it up.
===============
*/
int SNDDMA_GetDMAPos( void )
{
    MMTIME	mmtime;
    int		s;
    DWORD	dwWrite;

    if ( !dsound_init )
    {
        return 0;
    }

    mmtime.wType = TIME_SAMPLES;
    pDSBuf->lpVtbl->GetCurrentPosition(pDSBuf, &mmtime.u.sample, &dwWrite);

    s = mmtime.u.sample;

    s >>= sample16;

    s &= (dma.samples-1);

    return s;
}

/*
==============
SNDDMA_BeginPainting
 
Makes sure dma.buffer is valid
===============
*/
void SNDDMA_BeginPainting( void )
{
    int		reps;
    DWORD	dwSize2;
    DWORD	*pbuf, *pbuf2;
    HRESULT	hresult;
    DWORD	dwStatus;

    if ( !pDSBuf )
    {
        return;
    }

    // if the buffer was lost or stopped, restore it and/or restart it
    if ( pDSBuf->lpVtbl->GetStatus (pDSBuf, &dwStatus) != DS_OK )
    {
        Com_Printf ("Couldn't get sound buffer status\n");
    }

    if (dwStatus & DSBSTATUS_BUFFERLOST)
        pDSBuf->lpVtbl->Restore (pDSBuf);

    if (!(dwStatus & DSBSTATUS_PLAYING))
        pDSBuf->lpVtbl->Play(pDSBuf, 0, 0, DSBPLAY_LOOPING);

    // lock the dsound buffer

    reps = 0;
    dma.buffer = NULL;

    while ((hresult = pDSBuf->lpVtbl->Lock(pDSBuf, 0, gSndBufSize, &pbuf, &locksize,
                                           &pbuf2, &dwSize2, 0)) != DS_OK)
    {
        if (hresult != DSERR_BUFFERLOST)
        {
            Com_Printf( "SNDDMA_BeginPainting: Lock failed with error '%s'\n", DSoundError( hresult ) );
            S_Shutdown ();
            return;
        }
        else
        {
            pDSBuf->lpVtbl->Restore( pDSBuf );
        }

        if (++reps > 2)
            return;
    }
    dma.buffer = (unsigned char *)pbuf;
}

/*
==============
SNDDMA_Submit
 
Send sound to device if buffer isn't really the dma buffer
Also unlocks the dsound buffer
===============
*/
void SNDDMA_Submit( void )
{
    // unlock the dsound buffer
    if ( pDSBuf )
    {
        pDSBuf->lpVtbl->Unlock(pDSBuf, dma.buffer, locksize, NULL, 0);
    }
}


/*
=================
SNDDMA_Activate
 
When we change windows we need to do this
=================
*/
void SNDDMA_Activate( void )
{
    if ( !pDS )
    {
        return;
    }

    if ( DS_OK != pDS->lpVtbl->SetCooperativeLevel( pDS, g_wv.hWnd, DSSCL_PRIORITY ) )
    {
        Com_Printf ("sound SetCooperativeLevel failed\n");
        SNDDMA_Shutdown ();
    }
}


